home *** CD-ROM | disk | FTP | other *** search
/ Total Network Tools 2002 / NextStepPublishing-TotalNetworkTools2002-Win95.iso / Archive / Misc Servers / Zope.exe / ZDAEMON.PY < prev    next >
Encoding:
Python Source  |  1999-10-29  |  8.3 KB  |  263 lines
  1. #!/usr/bin/env python
  2. ##############################################################################
  3. # 
  4. # Zope Public License (ZPL) Version 1.0
  5. # -------------------------------------
  6. # 
  7. # Copyright (c) Digital Creations.  All rights reserved.
  8. # 
  9. # This license has been certified as Open Source(tm).
  10. # 
  11. # Redistribution and use in source and binary forms, with or without
  12. # modification, are permitted provided that the following conditions are
  13. # met:
  14. # 
  15. # 1. Redistributions in source code must retain the above copyright
  16. #    notice, this list of conditions, and the following disclaimer.
  17. # 
  18. # 2. Redistributions in binary form must reproduce the above copyright
  19. #    notice, this list of conditions, and the following disclaimer in
  20. #    the documentation and/or other materials provided with the
  21. #    distribution.
  22. # 
  23. # 3. Digital Creations requests that attribution be given to Zope
  24. #    in any manner possible. Zope includes a "Powered by Zope"
  25. #    button that is installed by default. While it is not a license
  26. #    violation to remove this button, it is requested that the
  27. #    attribution remain. A significant investment has been put
  28. #    into Zope, and this effort will continue if the Zope community
  29. #    continues to grow. This is one way to assure that growth.
  30. # 
  31. # 4. All advertising materials and documentation mentioning
  32. #    features derived from or use of this software must display
  33. #    the following acknowledgement:
  34. # 
  35. #      "This product includes software developed by Digital Creations
  36. #      for use in the Z Object Publishing Environment
  37. #      (http://www.zope.org/)."
  38. # 
  39. #    In the event that the product being advertised includes an
  40. #    intact Zope distribution (with copyright and license included)
  41. #    then this clause is waived.
  42. # 
  43. # 5. Names associated with Zope or Digital Creations must not be used to
  44. #    endorse or promote products derived from this software without
  45. #    prior written permission from Digital Creations.
  46. # 
  47. # 6. Modified redistributions of any form whatsoever must retain
  48. #    the following acknowledgment:
  49. # 
  50. #      "This product includes software developed by Digital Creations
  51. #      for use in the Z Object Publishing Environment
  52. #      (http://www.zope.org/)."
  53. # 
  54. #    Intact (re-)distributions of any official Zope release do not
  55. #    require an external acknowledgement.
  56. # 
  57. # 7. Modifications are encouraged but must be packaged separately as
  58. #    patches to official Zope releases.  Distributions that do not
  59. #    clearly separate the patches from the original work must be clearly
  60. #    labeled as unofficial distributions.  Modifications which do not
  61. #    carry the name Zope may be packaged in any form, as long as they
  62. #    conform to all of the clauses above.
  63. # 
  64. # 
  65. # Disclaimer
  66. # 
  67. #   THIS SOFTWARE IS PROVIDED BY DIGITAL CREATIONS ``AS IS'' AND ANY
  68. #   EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  69. #   IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  70. #   PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL DIGITAL CREATIONS OR ITS
  71. #   CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  72. #   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  73. #   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  74. #   USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  75. #   ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  76. #   OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  77. #   OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  78. #   SUCH DAMAGE.
  79. # 
  80. # 
  81. # This software consists of contributions made by Digital Creations and
  82. # many individuals on behalf of Digital Creations.  Specific
  83. # attributions are listed in the accompanying credits file.
  84. # 
  85. ##############################################################################
  86. """
  87.  
  88. zinit, slightly smarter server manager and ZServer startup script.
  89.  
  90.   zinit will:
  91.  
  92.     - Fork a parent and a child
  93.  
  94.     - restart the child if it dies
  95.  
  96.     - write a pid file so you can kill (the parent)
  97.  
  98.     - reports the childs pid to stdout so you can kill that too
  99.  
  100. TODO
  101.  
  102.   - Have the parent reap the children when it dies
  103.  
  104.   - etc.
  105.  
  106. """
  107.  
  108. import os, sys, time, signal, posix
  109.  
  110. pyth = sys.executable
  111.  
  112. class KidDiedOnMeError(Exception):
  113.     pass
  114.  
  115. class ExecError(Exception):
  116.     pass
  117.  
  118. class ForkError(Exception):
  119.     pass
  120.  
  121. FORK_ATTEMPTS = 2
  122.  
  123. # This is the number of seconds between the parent pulsing the child.
  124. # Set to 0 to deactivate pulsing.
  125.  
  126. BEAT_DELAY = 0
  127. VERBOSE = 1
  128.  
  129. # If you want the parent to 'pulse' Zope every 'BEAT_DELAY' seconds,
  130. # put the URL to the method you want to call here.  This can be any
  131. # methodish object that can be called through the web.  Format is:
  132. #
  133. # activities = (("http://x/method", "username", "password"),)
  134. #
  135. # username and password may be None if the method does not require
  136. # authentication. 
  137.  
  138. # activities = (('http://localhost:9222/Heart/heart', 'michel', '123'),
  139. #               )
  140.  
  141. import zLOG
  142.  
  143. #this is a bit of a hack so I dont have to change too much code
  144. def pstamp(message, sev):
  145.     zLOG.LOG("zdaemon", sev,
  146.              ("zdaemon: %s: %s" % (time.ctime(time.time()), message)))
  147.  
  148. def heartbeat():
  149.     print 'tha-thump'
  150.     if activities:
  151.         for a in activities:
  152.             try:
  153.                 result = ZPublisher.Client.call(a[0], a[1], a[2])
  154.             except:
  155.                 pstamp('activity %s failed!' % a[0], zLOG.WARNING)
  156.                 return
  157.  
  158.             if result and VERBOSE:
  159.                 pstamp('activity %s returned: %s' % (a[0], result),
  160.                        zLOG.BLATHER)
  161.  
  162.  
  163. def forkit(attempts = FORK_ATTEMPTS):
  164.     while attempts:
  165.         # if at first you don't succeed...
  166.         attempts = attempts - 1
  167.         try:
  168.             pid = os.fork()
  169.         except os.error:
  170.             pstamp('Houston, the fork failed', zLOG.ERROR)
  171.             time.sleep(2)
  172.         else:
  173.             pstamp('Houston, we have forked', zLOG.INFO)
  174.             return pid
  175.  
  176. def run(argv, pidfile=''):
  177.     if os.environ.has_key('ZDAEMON_MANAGED'):
  178.         # We're the child at this point.  Don't ask. :/
  179.  
  180.         return
  181.     
  182.     os.environ['ZDAEMON_MANAGED']='TRUE'
  183.     while 1:
  184.         if not os.environ.has_key('Z_DEBUG_MODE'):
  185.             pid = os.fork()
  186.             if pid:
  187.                 sys.exit(0)
  188.             posix.setsid()
  189.  
  190.         lastt=time.time()
  191.         try:
  192.             pid = forkit()
  193.  
  194.             if pid is None:
  195.                 raise ForkError
  196.  
  197.             elif pid:
  198.                 # Parent 
  199.                 pstamp(('Hi, I just forked off a kid: %s' % pid), zLOG.INFO)
  200.                 # here we want the pid of the parent
  201.                 if pidfile:
  202.                     pf = open(pidfile, 'w+')
  203.                     pf.write(("%s" % os.getpid()))
  204.                     pf.close()
  205.  
  206.                 while 1: 
  207.                     if not BEAT_DELAY:
  208.                         p,s = os.waitpid(pid, 0)
  209.                     else:
  210.                         p,s = os.waitpid(pid, os.WNOHANG)
  211.                         if not p:
  212.                             time.sleep(BEAT_DELAY)
  213.                             heartbeat()
  214.                             continue
  215.                     if s:
  216.                         pstamp(('Aiieee! %s exited with error code: %s' 
  217.                                 % (p, s)), zLOG.ERROR)
  218.                         if time.time()-lastt < 20:
  219.                             raise ForkError # We're probably hosed
  220.                     else:
  221.                         raise ForkError
  222.                         pstamp(('The kid, %s, died on me.' % pid),
  223.                                zLOG.WARNING)
  224.  
  225.                     raise KidDiedOnMeError
  226.  
  227.             else:
  228.                 # Child
  229.                 os.execv(pyth, (pyth,)+tuple(argv))
  230.                 
  231.  
  232.         except ExecError:
  233.             sys.exit()
  234.         except ForkError:
  235.             sys.exit()
  236.         except KidDiedOnMeError:
  237.             pass
  238.  
  239. def main():
  240.     argv=sys.argv[1:]
  241.     if argv and argv[0][:2]=='-p':
  242.         pidf=argv[0][2:]
  243.         del argv[0]
  244.     else:
  245.         pidf=''
  246.  
  247.     if not argv:
  248.         print __doc__ % vars()
  249.         print
  250.         print 'Error: no script given'
  251.             
  252.     run(argv, pidf)
  253.     
  254.  
  255. if __name__ == '__main__': main()
  256.  
  257.  
  258.  
  259.  
  260.  
  261.  
  262.  
  263.